/*
* Copyright 2022 Beijing Zitiao Network Technology Co., Ltd.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
*     http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/


package net.bytedance.security.app.taintflow

import net.bytedance.security.app.engineconfig.PointerPropagationConfig
import soot.SootMethod

/**
 * rule from the configuration file
 */
class DefaultPointerPropagationRule(cfg: PointerPropagationConfig) : IPointerFlowRule {
    private val methodRule: Map<String, List<FlowItem>>

    init {
        this.methodRule =
            cfg.methodNameRule.mapValues { it.value.toFlowItem() } + cfg.methodSigRule.mapValues { it.value.toFlowItem() }
    }

    override fun flow(calleeMethod: SootMethod): List<FlowItem>? {
        methodRule[calleeMethod.signature]?.let {
            return it
        }
        methodRule[calleeMethod.name]?.let {
            return it
        }
        return null
    }

}